Skip to content

[Auto Parallel] Add spmd rule No.19 for roi_align and roi_align_grad ops. - #72925

Merged
luotao1 merged 15 commits into
PaddlePaddle:developfrom
ooooo-create:add_spmd_roi_align
Jun 11, 2025
Merged

luotao1 merged 15 commits into
PaddlePaddle:developfrom
ooooo-create:add_spmd_roi_align

Conversation

@ooooo-create

@ooooo-create ooooo-create commented May 25, 2025 •

Copy link
Copy Markdown
Collaborator

PR Category

Auto Parallel

PR Types

New features

Description

  • 为 roi_align 和 roi_align_grad 算子增加切分推导规则。
  • 只有输入 x 的 C 轴可以切分,其他轴和其他输入都不能进行切分,(目前boxes 第0维的切分支持仅仅通过切分规则不能实现)

@paddle-bot

paddle-bot Bot commented May 25, 2025

Copy link
Copy Markdown

你的PR提交成功,感谢你对开源项目的贡献!
请关注后续CI自动化测试结果,详情请参考Paddle-CI手册。
Your PR has been submitted. Thanks for your contribution!
Please wait for the result of CI firstly. See Paddle CI Manual for details.

@paddle-bot paddle-bot Bot added the contributor External developers label May 25, 2025
jeff41404
jeff41404 previously approved these changes Jun 10, 2025

@jeff41404 jeff41404 left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

zyfncg
zyfncg previously approved these changes Jun 10, 2025
@luotao1

luotao1 commented Jun 10, 2025

Copy link
Copy Markdown
Collaborator

请解决下冲突

@ooooo-create
ooooo-create dismissed stale reviews from zyfncg and jeff41404 via b39ba2d June 10, 2025 06:32
@luotao1
luotao1 merged commit 9966297 into PaddlePaddle:develop Jun 11, 2025
shanjiang7 pushed a commit to shanjiang7/Paddle that referenced this pull request Jun 12, 2025
@ooooo-create
ooooo-create deleted the add_spmd_roi_align branch September 29, 2025 09:44
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

contributor External developers HappyOpenSource Pro 进阶版快乐开源活动,更具挑战性的任务 skip-ci: all skip-ci: approval

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants